import torch.nn.functional as F
import torch
input_ = torch.tensor([[1,2,0,3,1],
                       [0,1,2,3,1],
                       [1,2,1,0,0],
                       [2,3,1,2,0],
                       [1,4,2,1,0]])
kernel = torch.tensor([[1,2,1],
                       [0,1,0],
                       [2,1,0]])
print(input_.shape,"\t",kernel.shape)
input_ = torch.reshape(input_,(1,1,5,5))
kernel = torch.reshape(kernel,[1,1,3,3])
output = F.conv2d(input_,kernel,stride=1)
print(output)
output2 = F.conv2d(input_,kernel,stride=1,padding=1)
print(output2)
